Python How to create concise and efficient list